SpringBoot 中 MyBatis |
您所在的位置：网站首页 › mybatis plus 切换数据源 › SpringBoot 中 MyBatis |
SpringBoot 中 MyBatis
|
使用场景
当你的项目中使用到多个数据源或者需要在程序运行过程中动态的添加数据源时可以参考本文中的实现。这里使用的是dynamic-datasource-spring-boot-starter ，它是一个基于springboot的快速集成多数据源的启动器。具体文档：dynamic-datasource-spring-boot-starter官方文档。 Demo环境与版本 JDK1.8SpringBoot 2.6.6MyBatis-Plus 3.4.1dynamic-datasource-spring-boot-starter 3.4.1 pom.xml下面是完成的pom.xml文件内容 4.0.0 org.springframework.boot spring-boot-starter-parent 2.6.6 vip.huhailong dynamic-datasource-demo 0.0.1-SNAPSHOT dynamic-datasource-demo About dynamic datasource demo 1.8 org.springframework.boot spring-boot-starter-web mysql mysql-connector-java runtime com.baomidou mybatis-plus-boot-starter 3.4.1 com.baomidou dynamic-datasource-spring-boot-starter 3.4.1 org.projectlombok lombok true org.springframework.boot spring-boot-maven-plugin org.projectlombok lombok 情景1: 固定多数据源情景这种情景是多个数据源的IP地址、端口、用户名、密码等信息已知，可以写到application.yml的配置中，不会涉及到动态添加到场景，这种场景集成很简单。下面是application.yml配置文件内容： spring: datasource: dynamic: primary: master #设置默认的数据源，默认值为master strict: false #是否弃用严格模式，如果启用在味匹配到指定数据源时抛出异常 datasource: master: url: jdbc:mysql://192.168.1.110:3306/dynamic1 username: root password: hhl@MySQL_1121 driverClassName: com.mysql.cj.jdbc.Driver test: url: jdbc:mysql://192.168.1.110:3306/dynamic2 username: root password: hhl@MySQL_1121 driverClassName: com.mysql.cj.jdbc.Driver mybatis-plus: configuration: log-impl: org.apache.ibatis.logging.stdout.StdOutImpl server: port: 8080 servlet: context-path: /demo上面的配置中从dynamic开始是有关多数据源的配置，primary用来配置主数据源也叫默认数据源，它的默认值是master，因此，如果不指定primary的话一定要有名为master的数据源。strict 是用来配置是否启用严格 模式，如果为true表示启用，此时如果没有匹配的数据源会抛出异常，如果为false则会使用默认数据源。datasource下面就是配置多数据源的信息了。 后面的mybatis-plus配置我主要是用来说明集成这个不会对mybatis-plus对配置产生影响，该怎么用mybatis-plus就继续怎么用。 测试验证正如配置文件中的那样，我使用了两个不同的数据库来模拟多数据源，每个数据库中添加了一张表——test，如下： test表对应的实体类 package vip.huhailong.demo.entity; import com.baomidou.mybatisplus.annotation.TableName; import lombok.Data; @Data @TableName public class Test { private Integer id; private String name; } TestMapper.javaTestMapper 接口类 package vip.huhailong.demo.mapper; import com.baomidou.mybatisplus.core.mapper.BaseMapper; import vip.huhailong.demo.entity.Test; public interface TestMapper extends BaseMapper { } ITestService.java对应的接口类 package vip.huhailong.demo.service; import com.baomidou.mybatisplus.extension.service.IService; import vip.huhailong.demo.entity.Test; public interface ITestService extends IService { } TestServiceImpl.java对应接口的实现类 package vip.huhailong.demo.service.impl; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import org.springframework.stereotype.Service; import vip.huhailong.demo.entity.Test; import vip.huhailong.demo.mapper.TestMapper; import vip.huhailong.demo.service.ITestService; @Service public class TestServiceImpl extends ServiceImpl implements ITestService { } TestController.java接口类 package vip.huhailong.demo.controller; import com.baomidou.dynamic.datasource.DynamicRoutingDataSource; import com.baomidou.dynamic.datasource.annotation.DS; import com.baomidou.dynamic.datasource.creator.DefaultDataSourceCreator; import com.baomidou.dynamic.datasource.spring.boot.autoconfigure.DataSourceProperty; import org.springframework.beans.BeanUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.*; import vip.huhailong.demo.entity.DataSourceDTO; import vip.huhailong.demo.entity.Test; import vip.huhailong.demo.service.ITestService; import javax.sql.DataSource; import java.util.List; import java.util.Set; @RestController @RequestMapping("/test") public class TestController { @Autowired ITestService testService; @GetMapping("/get1") public List test1(){ return testService.list(); } @DS("test") @GetMapping("/get2") public List test2(){ return testService.list(); } @GetMapping("/now") public Set now(){ DynamicRoutingDataSource source = (DynamicRoutingDataSource) dataSource; return source.getDataSources().keySet(); } }上面代码中除了TestController.java外都大同小异，mybatis-plus都常规写法，也可以用代码生成器生成。这里主要看一下TestController.java这个类，这里面有三个接口： “/get1”：使用默认数据库，也就是dynamic1。“/get2”：使用test数据库，也就是dynamic2。“/now”：查询当前的所有数据源。 在第二个接口上面使用类注解@DS("test"),该注解就是用来切换数据源的，里面的值就是数据源的名称。下面看一下接口的调用情况。 可以看到和我们预想的一样。
情景2：动态数据源情景
改情景下的数据源可能是在运行过程中添加进去的，使用时我们在上面的固定多数据源的代码中进行修改实验。 首先将application.yml文件中的test数据源注释掉： spring: datasource: dynamic: primary: master #设置默认的数据源，默认值为master strict: false #是否弃用严格模式，如果启用在味匹配到指定数据源时抛出异常 datasource: master: url: jdbc:mysql://192.168.1.110:3306/dynamic1 username: root password: hhl@MySQL_1121 driverClassName: com.mysql.cj.jdbc.Driver # test: # url: jdbc:mysql://192.168.1.110:3306/dynamic2 # username: root # password: hhl@MySQL_1121 # driverClassName: com.mysql.cj.jdbc.Driver mybatis-plus: configuration: log-impl: org.apache.ibatis.logging.stdout.StdOutImpl server: port: 8080 servlet: context-path: /demo然后需要改变的文件是TestController.java这个文件，如下： package vip.huhailong.demo.controller; import com.baomidou.dynamic.datasource.DynamicRoutingDataSource; import com.baomidou.dynamic.datasource.annotation.DS; import com.baomidou.dynamic.datasource.creator.DefaultDataSourceCreator; import com.baomidou.dynamic.datasource.spring.boot.autoconfigure.DataSourceProperty; import org.springframework.beans.BeanUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.*; import vip.huhailong.demo.entity.DataSourceDTO; import vip.huhailong.demo.entity.Test; import vip.huhailong.demo.service.ITestService; import javax.sql.DataSource; import java.util.List; import java.util.Set; @RestController @RequestMapping("/test") public class TestController { @Autowired private DataSource dataSource; @Autowired private DefaultDataSourceCreator defaultDataSourceCreator; @Autowired ITestService testService; @GetMapping("/get1") public List test1(){ return testService.list(); } @DS("test") @GetMapping("/get2") public List test2(){ return testService.list(); } @GetMapping("/now") public Set now(){ DynamicRoutingDataSource source = (DynamicRoutingDataSource) dataSource; return source.getDataSources().keySet(); } @PostMapping("/addSource") public Set addSource(@RequestBody DataSourceDTO dto){ DataSourceProperty dataSourceProperty = new DataSourceProperty(); BeanUtils.copyProperties(dto,dataSourceProperty); DynamicRoutingDataSource source = (DynamicRoutingDataSource) dataSource; DataSource dataSource = defaultDataSourceCreator.createDataSource(dataSourceProperty); source.addDataSource(dto.getPoolName(),dataSource); return source.getDataSources().keySet(); } @DeleteMapping("/remove") public String removeDataSource(String name){ DynamicRoutingDataSource source = (DynamicRoutingDataSource) dataSource; source.removeDataSource(name); return "删除成功"; } }DataSourceDTO.java package vip.huhailong.demo.entity; import lombok.Data; @Data public class DataSourceDTO { private String poolName; private String driverClassName; private String url; private String username; private String password; }上面的DataSourceDTO.java类是用来接收数据源参数的。里面的poolName就理解为数据源名称就行 可以看到上面的代码中我们注入了DataSource和DefaultDataSourceCreator，这里DefaultDataSourceCreator 是一个通用的创建器，其根据环境自动选择连接池。除了它我们也可以选择其他： BasicDataSourceCreatorJndiDataSourceCreatorDruidDataSourceCreatorHikariDataSourceCreatorBeeCpDataSourceCreatorDbcp2DataSourceCreator 可以根据自己的不同需求选择上面的类型。 除此之外代码中还添加了俩个方法：addSource：添加数据源removeDataSource：移除数据源 重启项目，经过测试：当没有执行添加数据源接口时，请求"/get1"和"/get2"均返回的是数据源1的结果，执行"/now"时显示只有一个数据源，当执行了添加数据源接口后，执行结果和上面的固定多数据源执行结果相同。执行添加数据源的接口json参数如下： "poolName": "test", "url": "jdbc:mysql://192.168.1.110:3306/dynamic2", "username": "root", "password": "your_mysql_password", "driverClassName": "com.mysql.cj.jdbc.Driver" }以上就是一个springboot 中 MyBatis-Plus 集成动态多数据源的一个简单的Demo。需要注意的是，Demo中为了方便测试将@DS注解注解用到了Controller类中，这个推荐使用到接口的实现类中，虽然可以在Controller中使用，但是放到接口实现类中更合理和规范。 项目地址：dynamic-datasource-demo ❤️ ❤️ ❤️ ❤️ ❤️ 如果本文对你有一点点的帮助请点个大大赞👍 ❤️ ❤️ ❤️ ❤️ ❤️ |
然后在项目中建立一个实体类——Test.java，再像平时那样搞定对应的mapper文件、service文件以及对应的实现类，然后创建一个测试的接口类用来做测试，下面是响应的文件【本文地址】
公司简介
联系我们
今日新闻 |
点击排行 |
|
推荐新闻 |
图片新闻 |
|
专题文章 |